package com.woniumall.web;

import com.woniumall.pojo.Massage;
import com.woniumall.util.uploadcheck;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadBase;
import org.apache.commons.fileupload.ProgressListener;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

import javax.servlet.ServletException;
import javax.servlet.annotation.WebFilter;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.util.List;
import java.util.UUID;

/**
 * @auther linke
 * @create 2021-11-05 22:09
 */
@WebServlet("/upload/*")
public class uploadServlet extends BaseServlet {
    private String serverPath = null;
    private String realSavePath = null;
    private String fileExtName = null;
    private  String saveFilename = null;

    public void up(HttpServletRequest request, HttpServletResponse response) throws Exception {
     /*   response.setHeader("Content-type", "text/html;charset=UTF-8");
        response.setCharacterEncoding("UTF-8");*/
        serverPath = request.getRealPath("/");
        System.out.println(serverPath);
        String savePath = serverPath + "/uploads";
        String tmpPath = savePath + "/tmp";
        // Servlet初始化时执行,如果上传文件目录不存在则自动创建
        if (!new File(savePath).exists()) {
            new File(savePath).mkdirs();
        }
        if (!new File(tmpPath).exists()) {
            new File(tmpPath).mkdirs();
        }
        // 消息提示
        Massage message =new Massage();
        try {
            // 使用Apache文件上传组件处理文件上传步骤：
            // 1、创建一个DiskFileItemFactory工厂
            DiskFileItemFactory factory = new DiskFileItemFactory();
            // 设置工厂的缓冲区的大小，当上传的文件大小超过缓冲区的大小时，就会生成一个临时文件存放到指定的临时目录当中。
            factory.setSizeThreshold(1024 * 100);// 设置缓冲区的大小为100KB，如果不指定，那么缓冲区的大小默认是10KB
            // 设置上传时生成的临时文件的保存目录
            factory.setRepository(new File(tmpPath));
            // 2、创建一个文件上传解析器
            ServletFileUpload upload = new ServletFileUpload(factory);
            // 监听文件上传进度
            upload.setProgressListener(new ProgressListener() {
                public void update(long pBytesRead, long pContentLength, int arg2) {
                    System.out.println("文件大小为：" + pContentLength + ",当前已处理：" + pBytesRead);
                }
            });
            // 解决上传文件名的中文乱码
            upload.setHeaderEncoding("UTF-8");
            // 3、判断提交上来的数据是否是上传表单的数据
            if (!ServletFileUpload.isMultipartContent(request)) {
                // 按照传统方式获取数据
                return;
            }
            // 设置上传单个文件的大小的最大值，目前是设置为1024*1024字节，也就是1MB
            upload.setFileSizeMax(1024 * 1024);
            // 设置上传文件总量的最大值，最大值=同时上传的多个文件的大小的最大值的和，目前设置为10MB
            upload.setSizeMax(1024 * 1024 * 10);
            // 4、使用ServletFileUpload解析器解析上传数据，解析结果返回的是一个List<FileItem>集合，每一个FileItem对应一个Form表单的输入项
            List<FileItem> list = upload.parseRequest(request);
            for (FileItem item : list) {
                // 如果fileitem中封装的是普通输入项的数据
                if (item.isFormField()) {
                    String name = item.getFieldName();
                    // 解决普通输入项的数据的中文乱码问题
                    String value = item.getString("UTF-8");
                    // value = new String(value.getBytes("iso8859-1"),"UTF-8");
                    System.out.println(name + "=" + value);
                } else {// 如果fileitem中封装的是上传文件
                    String name = item.getFieldName();
                    // 得到上传的文件名称
                    String filename = item.getName();
                    System.out.println(name + "=" + filename);
                    if (filename == null || filename.trim().equals("")) {
                        continue;
                    }
                    // 注意：不同的浏览器提交的文件名是不一样的，有些浏览器提交上来的文件名是带有路径的，如： c:\a\b\1.txt，而有些只是单纯的文件名，如：1.txt
                    // 获取文件名
                    filename = filename.substring(filename.lastIndexOf("\\") + 1);
                    // 得到上传文件的扩展名
                    fileExtName = filename.substring(filename.lastIndexOf(".") + 1);
                    // 如果需要限制上传的文件类型，那么可以通过文件的扩展名来判断上传的文件类型是否合法
                    System.out.println("上传的文件的扩展名是：" + fileExtName);
                    // 获取item中的上传文件的输入流
                    InputStream in = item.getInputStream();
                    // 得到文件保存的名称（为防止文件覆盖的现象发生，要为上传文件产生一个唯一的文件名）
                    saveFilename = UUID.randomUUID().toString() + "_" + filename;
                    // 得到文件的保存目录
                    realSavePath = uploadcheck.makePath(saveFilename, savePath);
                    // 创建一个文件输出流
                    FileOutputStream out = new FileOutputStream(realSavePath + "\\" + saveFilename);
                    byte buffer[] = new byte[1024];
                    int len = 0;
                    while ((len = in.read(buffer)) > 0) {
                        out.write(buffer, 0, len);
                    }
                    in.close();
                    out.close();
                    // 删除处理文件上传时生成的临时文件
                    // item.delete();
                    message.setMassage("文件上传成功！");
                    message.setFlag(true);
                }
            }
            // 这里的异常处理比较简单。开发中请根据实际要求改写
        } catch (FileUploadBase.FileSizeLimitExceededException e) {
            message.setMassage("单个文件超出最大值！！！");
            message.setFlag(false);
            e.printStackTrace();
        } catch (FileUploadBase.SizeLimitExceededException e) {
            message.setMassage("上传文件的总的大小超出限制的最大值！！！");
            message.setFlag(false);
            e.printStackTrace();
        } catch (Exception e) {
            message.setMassage("文件上传失败！");
            message.setFlag(false);
            e.printStackTrace();
        }
      /*  PrintWriter writer = response.getWriter();
        writer.write(message);*/
        //System.out.println(saveFilename);
        //System.out.println(realSavePath+saveFilename);
        request.getSession().setAttribute("saveFilename",saveFilename);
        System.out.println(saveFilename);
        responseUtil(response,message);
    }


    public void down(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        this.doGet(req,resp);
    }
}
